from openai import OpenAI
from dotenv import load_dotenv  # 用于加载环境变量

load_dotenv()

client = OpenAI()


# OpenAi  embedding 模型 = "text-embedding-3-large"
# 千问： text-embedding-v3

def get_embeddings(text, model="text-embedding-v3"):
    # texts 是一个包含要获取嵌入表示的文本的列表
    data = client.embeddings.create(input=text, model=model).data
    print(data)
    return [record.embedding for record in data]


test_query = "你好"
vec = get_embeddings(test_query)
# 大模型 文本嵌入表示的列表
print(vec)
# 大模型 文本的嵌入表示
print(vec[0])
# 大模型文本的嵌入表示的纬度 1024
print(len(vec[0]))
